In [1]:
%matplotlib inline
import numpy as np
import matplotlib.pyplot as plt
import matplotlib.animation as animation
In [13]:
data1=np.load("0_100.npy")
data2=np.load("4000_4100.npy")
data3=np.load("8000_8100.npy")
data4=np.load("12000_12100.npy" )
data5=np.load("16000_16100.npy" )
data6=np.load("20000_20100.npy")
data7=np.load("24000_24100.npy" )
data8=np.load("28000_28100.npy")
trajectory=np.load("trajectory_epoch500_bias.npy")
world=np.load("world1.npy")

0~100

In [7]:
start=0
goal=start+100
hoge= np.sqrt(len(data1[0,0,start:goal]))
plt.figure(figsize=(15, 15), dpi=800)
for i in xrange(len(data1[0,0,start:goal])):
    plt.subplot(hoge+1,hoge+1 , i+1)
    fig = plt.imshow(data1[:,:,start+i])
    plt.axis('off')
In [17]:
plt.figure()
plt.imshow(world,"gray")
plt.scatter(trajectory[0,0:100]-14,trajectory[1,0:100]-14,color='y')
plt.scatter(trajectory[0,0]-14,trajectory[1,0]-14,color='red')
plt.axis('off')
plt.show()

4000~4100

In [18]:
start=0
goal=start+100
hoge= np.sqrt(len(data2[0,0,start:goal]))
plt.figure(figsize=(15, 15), dpi=800)
for i in xrange(len(data2[0,0,start:goal])):
    plt.subplot(hoge+1,hoge+1 , i+1)
    fig = plt.imshow(data2[:,:,start+i])
    plt.axis('off')
In [19]:
plt.figure()
plt.imshow(world,"gray")
plt.scatter(trajectory[0,4000:4100]-14,trajectory[1,4000:4100]-14,color='y')
plt.scatter(trajectory[0,4000]-14,trajectory[1,4000]-14,color='red')
plt.axis('off')
plt.show()

8000~8100

In [20]:
start=0
goal=start+100
hoge= np.sqrt(len(data3[0,0,start:goal]))
plt.figure(figsize=(15, 15), dpi=800)
for i in xrange(len(data3[0,0,start:goal])):
    plt.subplot(hoge+1,hoge+1 , i+1)
    fig = plt.imshow(data3[:,:,start+i])
    plt.axis('off')
In [21]:
plt.figure()
plt.imshow(world,"gray")
plt.scatter(trajectory[0,8000:8100]-14,trajectory[1,8000:8100]-14,color='y')
plt.scatter(trajectory[0,8000]-14,trajectory[1,8000]-14,color='red')
plt.axis('off')
plt.show()

12000~12100

In [22]:
start=0
goal=start+100
hoge= np.sqrt(len(data4[0,0,start:goal]))
plt.figure(figsize=(15, 15), dpi=800)
for i in xrange(len(data4[0,0,start:goal])):
    plt.subplot(hoge+1,hoge+1 , i+1)
    fig = plt.imshow(data4[:,:,start+i])
    plt.axis('off')
In [23]:
plt.figure()
plt.imshow(world,"gray")
plt.scatter(trajectory[0,12000:12100]-14,trajectory[1,12000:12100]-14,color='y')
plt.scatter(trajectory[0,12000]-14,trajectory[1,12000]-14,color='red')
plt.axis('off')
plt.show()

16000~16100

In [24]:
start=0
goal=start+100
hoge= np.sqrt(len(data5[0,0,start:goal]))
plt.figure(figsize=(15, 15), dpi=800)
for i in xrange(len(data5[0,0,start:goal])):
    plt.subplot(hoge+1,hoge+1 , i+1)
    fig = plt.imshow(data5[:,:,start+i])
    plt.axis('off')
In [25]:
plt.figure()
plt.imshow(world,"gray")
plt.scatter(trajectory[0,16000:16100]-14,trajectory[1,16000:16100]-14,color='y')
plt.scatter(trajectory[0,16000]-14,trajectory[1,16000]-14,color='red')
plt.axis('off')
plt.show()

20000~20100

In [26]:
start=0
goal=start+100
hoge= np.sqrt(len(data6[0,0,start:goal]))
plt.figure(figsize=(15, 15), dpi=800)
for i in xrange(len(data6[0,0,start:goal])):
    plt.subplot(hoge+1,hoge+1 , i+1)
    fig = plt.imshow(data6[:,:,start+i])
    plt.axis('off')
In [27]:
plt.figure()
plt.imshow(world,"gray")
plt.scatter(trajectory[0,20000:20100]-14,trajectory[1,20000:20100]-14,color='y')
plt.scatter(trajectory[0,20000]-14,trajectory[1,20000]-14,color='red')
plt.axis('off')
plt.show()
In [78]:
plt.imshow(world)
Out[78]:
<matplotlib.image.AxesImage at 0x13adc6fd0>
In [79]:
plt.figure()
plt.imshow(world,"gray")
plt.scatter(trajectory[0,:i]-14,trajectory[1,:i]-14,color='g')
plt.scatter(trajectory[0,0]-14,trajectory[1,0]-14,color='red')
plt.axis('off')
plt.show()
In [60]:
i=4000

plt.scatter(trajectory[0,-i:],trajectory[1,-i:], )
#plt.xlim([0, 28*3])
#plt.ylim([0,28*3])
Out[60]:
<matplotlib.collections.PathCollection at 0x112ce2ed0>
In [76]:
hoge=5*np.array((np.cos(1002/10.),np.cos(1002/10.)))
print hoge 

[ 4.72864615  4.72864615]
In [57]:
images[:,:,1]+hoge.T

---------------------------------------------------------------------------
ValueError                                Traceback (most recent call last)
<ipython-input-57-c68216720c60> in <module>()
----> 1 images[:,:,1]+hoge.T

ValueError: operands could not be broadcast together with shapes (28,28) (2,)
In [ ]: